package com.whut.common;

import java.util.Random;

/**
 * 生成验证码的工具类
 */
public class ValidateCodeUtils {
    /**
     * 随机生成验证码
     * @param length 长度为4位或6位
     * @return
     */
    public static Integer generateValidateCode(int length) {
        Integer code = null;
        if(length == 4) {
            code = new Random().nextInt(9999); //最大9999
            if(code < 1000) {
                code = code + 1000;   // 保证随机数四位
            }
        }
        else if(length == 6) {
            code = new Random().nextInt(999999);
            if(code < 100000) {
                code = code + 100000; // 保证随机数为六位数字
            }
        }
        else {
            throw new RuntimeException("只能生成4或6位验证码");
        }
        return code;
    }

    /**
     * 随机生成指定长度的字符串验证码
     * @param length 长度
     * @return
     */
    public static String generateValidate4String(int length) {
        Random random = new Random();
        String hash1 = Integer.toHexString(random.nextInt(1));
        String capstr = hash1.substring(0,length);
        return capstr;
    }
}

